//
//  walterFlowLayout.h
//  CollectionFlow
//
//  Created by bj_szd on 2025/5/6.
//

#import <UIKit/UIKit.h>

NS_ASSUME_NONNULL_BEGIN

@protocol WalterFlowLayoutDelegate <NSObject>
// 自定义一个协议，返回图片高度
-(CGFloat)collectionView:(UICollectionView * _Nonnull)collectionView layout:(UICollectionViewLayout *_Nonnull)collectionViewLayout width:(CGFloat)width heightForItemAtIndexPath:(nonnull NSIndexPath *)indexPath;

@end

@interface WalterFlowLayout : UICollectionViewLayout

@property (nonatomic,assign) int numberOfColumns;// 列数
@property (nonatomic,assign) id <WalterFlowLayoutDelegate> _Nonnull delegate;

@end

NS_ASSUME_NONNULL_END
